package com.example.ClassLoader;

import java.io.File;
import java.lang.reflect.Modifier;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;

public class GetImplementedCls {
    /**
     * 获取：1）接口的所有实现类；2）所有子类
     */
    public static void main(String[] args) throws Exception {
        // List<Class> clazzs = getAllInterfaceAchieveClass(LayerAble.class);
        // List<Class> clazzs = getAllSubClass(Function.class);
        // for(Class clazz : clazzs){
        //     System.out.println(clazz.getSimpleName() + "\t" + clazz.getName());
        // }
    }

    /**
     * 获取所有接口的实现类
     */
    @SuppressWarnings("unchecked")
    public static List<Class> getAllInterfaceAchieveClass(Class clazz) throws Throwable {
        ArrayList<Class> list = new ArrayList<>();
        // 判断是否是接口
        if (clazz.isInterface()) {
            ArrayList<Class> allClass = getAllClassByPath(clazz.getPackage().getName());
            /**
             * 循环判断路径下的所有类是否实现了指定的接口 并且排除接口类自己
             */
            for (int i = 0; i < allClass.size(); i++) {
                // 排除抽象类
                if (Modifier.isAbstract(allClass.get(i).getModifiers())) {
                    continue;
                }
                // 判断是不是同一个接口
                if (clazz.isAssignableFrom(allClass.get(i))) {
                    if (!clazz.equals(allClass.get(i))) {
                        list.add(allClass.get(i));
                    }
                }
            }
        }
        return list;
    }
    /**
     * 获取所有子类
     */
    public static List<Class> getAllSubClass(Class clazz) throws Throwable {
        ArrayList<Class> list = new ArrayList<>();
        ArrayList<Class> allClass = getAllClassByPath(clazz.getPackage().getName());
        for (Class aClass : allClass) {
            // 排除抽象类
            if (Modifier.isAbstract(aClass.getModifiers())) {
                continue;
            }
            // 判断是不是同一个接口
            if (clazz.isAssignableFrom(aClass)) {
                if (!clazz.equals(aClass)) {
                    list.add(aClass);
                }
            }
        }
        return list;
    }

    /**
     * 从指定路径下获取所有类
     */
    public static ArrayList<Class> getAllClassByPath(String packagename) throws Exception {
        ArrayList<Class> list = new ArrayList<>();
        ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
        String path = packagename.replace('.', '/');
        ArrayList<File> fileList = new ArrayList<>();
        Enumeration<URL> enumeration = classLoader.getResources(path);
        while (enumeration.hasMoreElements()) {
            URL url = enumeration.nextElement();
            fileList.add(new File(url.getFile()));
        }
        for (File file : fileList) {
            list.addAll(findClass(file, packagename));
        }
        return list;
    }

    /**
     * 如果file是文件夹，则递归调用findClass方法，或者文件夹下的类 如果file本身是类文件，则加入list中进行保存，并返回
     *
     * @param file
     * @param packagename
     */
    private static ArrayList<Class> findClass(File file, String packagename) throws Exception {
        ArrayList<Class> list = new ArrayList<>();
        if (!file.exists()) {
            return list;
        }
        File[] files = file.listFiles();
        for (File file2 : files) {
            if (file2.isDirectory()) {
                assert !file2.getName().contains(".");// 添加断言用于判断
                ArrayList<Class> arrayList = findClass(file2, packagename + "." + file2.getName());
                list.addAll(arrayList);
            } else if (file2.getName().endsWith(".class")) {
                // 保存的类文件不需要后缀.class
                list.add(Class.forName(packagename + '.' + file2.getName().substring(0, file2.getName().length() - 6)));
            }
        }
        return list;
    }
}
